added assertion into unit tests