diff --git a/tests/memory/oceandb.py b/tests/memory/oceandb.py
index 67879ce6..3e31afab 100644
--- a/tests/memory/oceandb.py
+++ b/tests/memory/oceandb.py
@@ -1,6 +1,6 @@
 import pytest
 from unittest.mock import Mock, patch
-from swarms.memory.oceandb import OceanDB
+from swarms.memory.ocean import OceanDB
 
 
 def test_init():