diff --git a/setup.py b/setup.py
index a045b76..fc33aeb 100644
--- a/setup.py
+++ b/setup.py
@@ -55,7 +55,7 @@ torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]
 
 def get_extensions():
     this_dir = os.path.dirname(os.path.abspath(__file__))
-    extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc")
+    extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc","datasets")
 
     main_source = os.path.join(extensions_dir, "vision.cpp")
     sources = glob.glob(os.path.join(extensions_dir, "**", "*.cpp"))