Fix warning code