diff --git a/gui.py b/gui.py
index b976e88..1d37d11 100644
--- a/gui.py
+++ b/gui.py
@@ -137,7 +137,7 @@ class GUI_PyGame:
 
         # Store screen and size
         self.size = size
-        self.screen = pygame.display.set_mode(self.size, pygame.FULLSCREEN)
+        self.screen = pygame.display.set_mode(size, pygame.FULLSCREEN)
 
         # Clear screen
         self.clear()
diff --git a/photobooth.py b/photobooth.py
index 96daba1..d183d2e 100755
--- a/photobooth.py
+++ b/photobooth.py
@@ -255,7 +255,7 @@ class Photobooth:
 #################
 
 def main():
-    photobooth = Photobooth(picture_basename, display_size, image_size, pose_time, display_time, 
+    photobooth = Photobooth(display_size, picture_basename, image_size, pose_time, display_time, 
                             gpio_trigger_channel, gpio_shutdown_channel, gpio_lamp_channel)
     photobooth.run()
     photobooth.teardown()