diff --git a/__init__.py b/__init__.py
index dff84d7..74e6454 100644
--- a/__init__.py
+++ b/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from timm_image import TimmImage
+from .timm_image import TimmImage
 
 def timm(**kwargs):
     return TimmImage(**kwargs)