fixed the test :)