tidied unit tests for ImageController