1 files changed, 2 insertions, 24 deletions

diff --git a/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java b/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java
index 4c3e52f3..e5115352 100644
--- a/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java
+++ b/subprojects/generator/src/main/java/tools/refinery/generator/ModelGenerator.java
@@ -10,10 +10,7 @@ import tools.refinery.language.model.problem.Relation;
 import tools.refinery.store.dse.strategy.BestFirstStoreManager;
 import tools.refinery.store.dse.transition.VersionWithObjectiveValue;
 import tools.refinery.store.map.Version;
-import tools.refinery.store.model.Model;
 import tools.refinery.store.model.ModelStore;
-import tools.refinery.store.reasoning.ReasoningAdapter;
-import tools.refinery.store.reasoning.ReasoningStoreAdapter;
 import tools.refinery.store.reasoning.interpretation.PartialInterpretation;
 import tools.refinery.store.reasoning.literal.Concreteness;
 import tools.refinery.store.reasoning.representation.PartialSymbol;
@@ -22,35 +19,16 @@ import tools.refinery.store.representation.TruthValue;
 
 import java.util.Collection;
 
-public class ModelGenerator {
-        private final ProblemTrace problemTrace;
-        private final ModelStore store;
-        private final Model model;
-        private final ReasoningAdapter reasoningAdapter;
+public class ModelGenerator extends AbstractRefinery {
         private final Version initialVersion;
 
         private int randomSeed = 1;
 
         public ModelGenerator(ProblemTrace problemTrace, ModelStore store, ModelSeed modelSeed) {
-                this.problemTrace = problemTrace;
-                this.store = store;
-                model = store.getAdapter(ReasoningStoreAdapter.class).createInitialModel(modelSeed);
-                reasoningAdapter = model.getAdapter(ReasoningAdapter.class);
+                super(problemTrace, store, modelSeed);
                 initialVersion = model.commit();
         }
 
-        public ProblemTrace getProblemTrace() {
-                return problemTrace;
-        }
-
-        public ModelStore getModelStore() {
-                return store;
-        }
-
-        public Model getModel() {
-                return model;
-        }
-
         public int getRandomSeed() {
                 return randomSeed;
         }