diff --git a/FastGeodis/__init__.py b/FastGeodis/__init__.py
index b7e5db4..b677288 100644
--- a/FastGeodis/__init__.py
+++ b/FastGeodis/__init__.py
@@ -30,7 +30,7 @@
 
 from typing import List
 import torch
-from _version import __version__
+from ._version import __version__
 import FastGeodisCpp