Fix for images controller