Fixing test (fixing the build)