Changes register_dataset signature