diff --git a/prettymaps/draw.py b/prettymaps/draw.py
index 068a722..6d715a1 100644
--- a/prettymaps/draw.py
+++ b/prettymaps/draw.py
@@ -3,6 +3,7 @@ from collections.abc import Iterable
 
 import osmnx as ox
 import pandas as pd
+from geopandas import GeoDataFrame
 import numpy as np
 from numpy.random import choice
 from shapely.geometry import Polygon, MultiPolygon, MultiLineString, GeometryCollection