1 files changed, 6 insertions, 4 deletions

diff --git a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/context/RelationalEngineContext.java b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/context/RelationalEngineContext.java
index 4eb8898b..3bad01b9 100644
--- a/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/context/RelationalEngineContext.java
+++ b/subprojects/store-query-viatra/src/main/java/tools/refinery/store/query/viatra/internal/context/RelationalEngineContext.java
@@ -2,17 +2,19 @@ package tools.refinery.store.query.viatra.internal.context;
 
 import org.eclipse.viatra.query.runtime.api.scope.IBaseIndex;
 import org.eclipse.viatra.query.runtime.api.scope.IEngineContext;
+import org.eclipse.viatra.query.runtime.matchers.context.IInputKey;
 import org.eclipse.viatra.query.runtime.matchers.context.IQueryRuntimeContext;
-
 import tools.refinery.store.model.Model;
-import tools.refinery.store.query.viatra.internal.update.ModelUpdateListener;
+import tools.refinery.store.query.view.AnyRelationView;
+
+import java.util.Map;
 
 public class RelationalEngineContext implements IEngineContext {
         private final IBaseIndex baseIndex = new DummyBaseIndexer();
         private final RelationalRuntimeContext runtimeContext;
 
-        public RelationalEngineContext(Model model, ModelUpdateListener updateListener) {
-                runtimeContext = new RelationalRuntimeContext(model, updateListener);
+        public RelationalEngineContext(Model model, Map< AnyRelationView, IInputKey> inputKeys) {
+                runtimeContext = new RelationalRuntimeContext(model, inputKeys);
         }
 
         @Override