Small tweak, ensure image size