diff --git a/scripts/plot_NN_2D.py b/scripts/plot_NN_2D.py
index 112c1bf5dcf5fee8523060cc193df6af7bef79ce..04016cbd396e9e22ca2f3d76ab9dc5f167911a24 100755
--- a/scripts/plot_NN_2D.py
+++ b/scripts/plot_NN_2D.py
@@ -129,7 +129,7 @@ if args.mode.startswith("mean"):
         if hasattr(c, "get_input_list"):
             transform_function = lambda inp : c.get_input_list(c.scaler.transform(inp))
         else:
-            transform_function = c.scaler.transform(inp)
+            transform_function = c.scaler.transform
         plot_NN_vs_var_2D_all(
             args.output_filename,
             means=means,