Image class update

Failed to catch image class mismatch in last update.