Controller error fixed