Fix the one measly unit test (not a good start...)