diff --git a/geolib/__init__.py b/geolib/__init__.py
index 75787af3..b2a8e72c 100644
--- a/geolib/__init__.py
+++ b/geolib/__init__.py
@@ -3,3 +3,5 @@
 """
 
 __version__ = "2.4.1"
+
+from .models import *
diff --git a/geolib/models/dstability/dstability_model.py b/geolib/models/dstability/dstability_model.py
index cff727be..8bd7810b 100644
--- a/geolib/models/dstability/dstability_model.py
+++ b/geolib/models/dstability/dstability_model.py
@@ -5,7 +5,7 @@
 
 import matplotlib.pyplot as plt
 from pydantic import DirectoryPath, FilePath
-from shapely.geometry import LineString, Point, Polygon
+from shapely.geometry import LineString, Polygon
 from shapely.ops import polygonize
 from shapely.validation import make_valid