Change expectation to fix error in last commit