diff --git a/src/picamui/picamui.py b/src/picamui/picamui.py
index dc34c95..b99718b 100644
--- a/src/picamui/picamui.py
+++ b/src/picamui/picamui.py
@@ -41,7 +41,7 @@ class PiCamUi:
         self.uiElements.append(btnExit)
 
     def updatePreview(self, rgb, res):
-        img = pygame.image.frombuffer(rgb[0:(res[0] * res[1] * 3)], res, 'RGB')
+        img = pygame.image.frombuffer(rgb, res, 'RGB')
         img = pygame.transform.scale(img, (self.screen.get_width(), self.screen.get_height()))
 
         if img: