fixing some code which only took the small image.