Fix Compilation issue