fixing last test