Small changes in assert