Fixed tests non-compile.