diff --git a/UIImage Scanline Floodfill/UIImage Scanline Floodfill/UIImage+FloodFill.m b/UIImage Scanline Floodfill/UIImage Scanline Floodfill/UIImage+FloodFill.m
index f74a47b..3f45a9d 100755
--- a/UIImage Scanline Floodfill/UIImage Scanline Floodfill/UIImage+FloodFill.m	
+++ b/UIImage Scanline Floodfill/UIImage Scanline Floodfill/UIImage+FloodFill.m	
@@ -49,7 +49,7 @@ - (UIImage *) floodFillFromPoint:(CGPoint)startPoint withColor:(UIColor *)newCol
         NSUInteger bytesPerRow = CGImageGetBytesPerRow(imageRef);
         NSUInteger bitsPerComponent = CGImageGetBitsPerComponent(imageRef);
 
-        unsigned char *imageData = malloc(height * width * bytesPerPixel);
+        unsigned char *imageData = malloc(height * bytesPerRow);
       
         CGBitmapInfo bitmapInfo = CGImageGetBitmapInfo(imageRef);