Added an assert to  tests