Fix for image controller.