Added new unit test which made me add another method.