Fixed bug when changing key for controller