Fixed non compiling code.