diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 9105af4d2e4bbc38c5ef74b5d39f3f5e646be418..deb471a9ab458903387ded3091c7a9331214ce33 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2550,7 +2550,7 @@ class NetworkInference(BaseConfig):
                                 accuracy=True,
                                 state=batch_name,
                                 plot_path=self.scenes_path,
-                                **self.kwargs)
+                                **self.plot_kwargs)
 
         return output