Fix dataset registration