Added ugly unit test