update feature to fix gallery