Use correct assertions on tests.