fixing the compilation error on test