diff --git a/imports.py b/imports.py
index 1bde5447..a9ef4b96 100644
--- a/imports.py
+++ b/imports.py
@@ -1,3 +1,5 @@
+from __future__ import with_statement
+
 import os
 import pkgutil
 import imp
diff --git a/test/completion/invalid.py b/test/completion/invalid.py
index b6e6b2e0..8afe20b3 100644
--- a/test/completion/invalid.py
+++ b/test/completion/invalid.py
@@ -1,5 +1,5 @@
-#? ['message']
-AttributeError()).message
+#? ['upper']
+str()).upper
 
 from a import (b
 def blub():