Python: Fix wrong Dataset initialization