Add missing wp_reset_query()