Fixing broken testsz