This can be just `return if dataset.class != Dataset`