Use correct assertions