Tests now green. Test fixed with correct assertion.