Reworded unit test