diff --git a/examples/SquareTest_2D/run.sh b/examples/SquareTest_2D/run.sh
index 242f1b8b729979b026cfe31002e84cd9ef741129..7d77e9c5bd89732970b47feb3a297ef92b345a01 100755
--- a/examples/SquareTest_2D/run.sh
+++ b/examples/SquareTest_2D/run.sh
@@ -11,4 +11,4 @@ fi
 ../swift -s -t 1 square.yml 2>&1 | tee output.log
 
 # Plot the solution
-python plotSolution.py 40
+python plotSolution.py 5