Fix image diff tests