Fix image check