Fixing complie issue