Fix assert
My solution