diff --git a/src/python/photon.py b/src/python/photon.py
index a49c677..16fcc91 100644
--- a/src/python/photon.py
+++ b/src/python/photon.py
@@ -23,6 +23,6 @@
 
 
 ani = animation.FuncAnimation(
-    fig, animate, interval=1000/framerate, blit=True, save_count = framerate * totalTime)
+    fig, animate, interval=1000/framerate, blit=True, save_count = int(framerate * totalTime))
 ani.save(f"{outputFolder}photon.mp4")
 # plt.show()

diff --git a/src/python/photon.py b/src/python/photon.py
index a49c677..16fcc91 100644
--- a/src/python/photon.py
+++ b/src/python/photon.py
@@ -23,6 +23,6 @@
 
 
 ani = animation.FuncAnimation(
-    fig, animate, interval=1000/framerate, blit=True, save_count = framerate * totalTime)
+    fig, animate, interval=1000/framerate, blit=True, save_count = int(framerate * totalTime))
 ani.save(f"{outputFolder}photon.mp4")
 # plt.show()
diff --git a/src/python/travellingWave.py b/src/python/travellingWave.py
index 4eda596..ad9d730 100644
--- a/src/python/travellingWave.py
+++ b/src/python/travellingWave.py
@@ -25,6 +25,6 @@
 
 
 ani = animation.FuncAnimation(
-    fig, animate, interval=1000/framerate, blit=True, save_count = framerate * totalTime)
+    fig, animate, interval=1000/framerate, blit=True, save_count = int(framerate * totalTime))
 ani.save(f"{outputFolder}travelling.mp4")
 # plt.show()