Fix that funky test