Fix dataset update behavior