diff --git a/frappy_psi/sea.py b/frappy_psi/sea.py
index 78479ecc..cd1e2c0d 100644
--- a/frappy_psi/sea.py
+++ b/frappy_psi/sea.py
@@ -42,7 +42,8 @@ from frappy.datatypes import ArrayOf, BoolType, \
 from frappy.errors import ConfigError, HardwareError, secop_error, CommunicationFailedError
 from frappy.lib import generalConfig, mkthread
 from frappy.lib.asynconn import AsynConn, ConnectionClosed
-from frappy.modules import Attached, Command, Done, Drivable, \
+from frappy.modulebase import Done
+from frappy.modules import Attached, Command, Drivable, \
     Module, Parameter, Property, Readable, Writable
 from frappy.protocol.dispatcher import make_update