Replace assert with should