Got unit test passing again