Fixed unit test that wasn't passing.