Fix get_gallery method