Fixes compilation error of tests