Fix up warning