diff --git a/neuralogic/core/builder/builder.py b/neuralogic/core/builder/builder.py
index 4a8c738f..c092ff38 100644
--- a/neuralogic/core/builder/builder.py
+++ b/neuralogic/core/builder/builder.py
@@ -82,10 +82,10 @@ def _build_samples(
         source_pipeline.execute(None if sources is None else sources.sources)
         java_model = source_pipeline.get()
 
-        groundings = java_model.r.collect(self.collectors.toList())
+        # groundings = java_model.r.collect(self.collectors.toList())
         logic_samples = java_model.s.collect(self.collectors.toList())
 
-        return [RawSample(sample, grounding) for sample, grounding in zip(logic_samples, groundings)]
+        return [RawSample(sample, None) for sample in logic_samples]
 
     def build_model(self, parsed_template, settings: SettingsProxy):
         neural_model = self.neural_model(parsed_template.getAllWeights(), settings.settings)
diff --git a/neuralogic/core/builder/components.py b/neuralogic/core/builder/components.py
index b06d45c2..00d95078 100644
--- a/neuralogic/core/builder/components.py
+++ b/neuralogic/core/builder/components.py
@@ -78,16 +78,19 @@ def draw_grounding(
         *args,
         **kwargs,
     ):
-        return draw_grounding(
-            self.grounding, filename, draw_ipython, img_type, value_detail, graphviz_path, *args, **kwargs
-        )
+        if False:
+            return draw_grounding(
+                self.grounding, filename, draw_ipython, img_type, value_detail, graphviz_path, *args, **kwargs
+            )
+
+        raise NotImplementedError
 
 
 class Sample(RawSample):
     __slots__ = ("id", "target", "neurons", "output_neuron", "java_sample")
 
     def __init__(self, sample, java_sample):
-        super().__init__(sample)
+        super().__init__(sample, None)
         serialized_sample = json.loads(str(sample.exportToJson()))
 
         self.id = serialized_sample["id"]