Updated image related tests