Fixed minor warning bug