Update all checks to use assertions