fixing tests
adding assert
fixing comment