Updated the location of assert