Fix more bugs with player movement