Forgot to add shutdown back