Fixing compiltation issue