Use correct assertion