Fix onRemove error