Fix image remove migration