Fixing the compile error