Fix unit tests not compiling