Update tests for removed image fields