Fix last test failure