diff --git a/flask_autodoc/__init__.py b/flask_autodoc/__init__.py
index 89e047e..6605939 100644
--- a/flask_autodoc/__init__.py
+++ b/flask_autodoc/__init__.py
@@ -1,3 +1,3 @@
 __author__ = 'arnaud'
 
-from flask.ext.autodoc.autodoc import Autodoc
+from .autodoc import Autodoc
diff --git a/tests/test_autodoc.py b/tests/test_autodoc.py
index e839c2f..8ae16f8 100644
--- a/tests/test_autodoc.py
+++ b/tests/test_autodoc.py
@@ -5,7 +5,7 @@
 import os
 
 from flask import Flask
-from flask.ext.autodoc import Autodoc
+from flask_autodoc import Autodoc
 
 
 class TestAutodoc(unittest.TestCase):