diff --git a/src/main/java/li/strolch/testbase/runtime/RuntimeMock.java b/src/main/java/li/strolch/testbase/runtime/RuntimeMock.java
index 76406c73e..58999dcb2 100644
--- a/src/main/java/li/strolch/testbase/runtime/RuntimeMock.java
+++ b/src/main/java/li/strolch/testbase/runtime/RuntimeMock.java
@@ -19,8 +19,8 @@ import java.io.File;
 import java.text.MessageFormat;
 
 import li.strolch.persistence.api.PersistenceHandler;
-import li.strolch.runtime.agent.ComponentContainer;
-import li.strolch.runtime.agent.StrolchAgent;
+import li.strolch.runtime.agent.api.ComponentContainer;
+import li.strolch.runtime.agent.api.StrolchAgent;
 import li.strolch.runtime.configuration.RuntimeConfiguration;
 import li.strolch.runtime.privilege.StrolchPrivilegeHandler;