Yet another test fix