added assert to unit test