Fixing test compiling