Add unit test to expose forgotten method