Fix test with wrong expectations