Fix test failure (stupid mistake)