fixing the last test