Fixing the code to compile.