Fixed the compilation errors!