Fixed tests not compiling