Removed .training_unit added accidentally on previous commit