Add hook in image mode to reload when return is pressed