Restore flying when using /survival