Remove unnecessary expect
Fix test based on previous update
All tests work
All tests work