Refactors test
Refactors tests