Fixed my tests. Again. Thanks @lpinca