Controller unit test changed slightly