updated assert
updated comment