diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py
index e7a03ba897..4b5961ac2d 100644
--- a/pytorch_lightning/__init__.py
+++ b/pytorch_lightning/__init__.py
@@ -1,3 +1,3 @@
-from .models import Trainer
+from .models.trainer import Trainer
 from .root_module.root_module import LightningModule
 from .root_module.decorators import data_loader
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 51658c2ec3..aed7c68040 100755
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@ setup(
     install_requires=[
         "torch>=1.1.0",
         "tqdm",
-        "test-tube==0.6.7.4",
+        "test-tube>=0.6.7.6",
     ],
     packages=find_packages(),
     long_description=open("README.md", encoding="utf-8").read(),