diff --git a/scripts/profile_output.py b/scripts/profile_output.py
index 528def6f..27b9e9f5 100755
--- a/scripts/profile_output.py
+++ b/scripts/profile_output.py
@@ -1,4 +1,5 @@
 #!/usr/bin/env python3.6
+# -*- coding: utf-8 -*-
 """
 Profile a piece of Python code with ``profile``. Tries a completion on a
 certain piece of code.
@@ -18,7 +19,11 @@ Options:
 """
 
 import time
-import profile
+try:
+    # For Python 2
+    import cProfile as profile
+except ImportError:
+    import profile
 import pstats
 
 from docopt import docopt