diff --git a/stream_topic/NAM/NAM.py b/stream_topic/NAM/NAM.py
index 89525febc8..92d1b480d6 100644
--- a/stream_topic/NAM/NAM.py
+++ b/stream_topic/NAM/NAM.py
@@ -323,6 +323,9 @@ def __init__(
             columns=["text", "tokens"], errors="ignore"
         )
 
+        if "predictions" in self.structured_data.columns:
+            self.structured_data = self.structured_data.drop(columns=["predictions"])
+
         self.target_column = target_column
 
         # Combine topic probabilities with structured data
@@ -520,7 +523,6 @@ def forward(self, x):
 
     def training_step(self, batch, batch_idx):
         x, y = batch
-        print(y)
         y_hat = self(x)
         loss = self.loss_fn(y_hat, y)
 
diff --git a/stream_topic/__version__.py b/stream_topic/__version__.py
index ba9f224e26..277ccdab5b 100644
--- a/stream_topic/__version__.py
+++ b/stream_topic/__version__.py
@@ -1,4 +1,4 @@
 """Version information."""
 
 # The following line *must* be the last in the module, exactly as formatted:
-__version__ = "0.1.7"
+__version__ = "0.1.8"