Fixed an error in one of my tests