Fixed one of my tests