diff --git a/celeste/plot.py b/celeste/plot.py
index 84dd7b3..a9314f4 100644
--- a/celeste/plot.py
+++ b/celeste/plot.py
@@ -8,16 +8,13 @@ from multiprocessing import Pool
 
 m = Path("model_data/current")
 
-scaled = True
-
 # Make "predicted reward" plots
 def plot_pred(src_model):
 	plotting.predicted_reward(
 		src_model,
 		m / f"plots/predicted/{src_model.stem}.png",
 
-		device = torch.device("cpu"),
-		scaled = scaled
+		device = torch.device("cpu")
 	)
 
 # Make "best action" plots
@@ -26,8 +23,7 @@ def plot_best(src_model):
 		src_model,
 		m / f"plots/best_action/{src_model.stem}.png",
 
-		device = torch.device("cpu"),
-		scaled = scaled
+		device = torch.device("cpu")
 	)
 
 # Make "actual reward" plots