Quick fix to unexpected image gain problem