Covering edge case on unit test