One last test fix