(Hopefully) fixing broken unit test