diff --git a/tests/context.py b/tests/context.py
index c4096ee..0b7ffe3 100644
--- a/tests/context.py
+++ b/tests/context.py
@@ -2,5 +2,5 @@ import os
 import sys
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
 
-import sample
+import bonob
 
diff --git a/tests/test_one.py b/tests/test_one.py
index 53b3337..cb53f92 100644
--- a/tests/test_one.py
+++ b/tests/test_one.py
@@ -1,7 +1,5 @@
-#from context import bonob
+from context import bonob
 
-
-# content of test_sample.py
 def f(x):
     return x + 1