Fix bad test assertion