Fixing test
Fixing last test