         if unitSelectorContainer?
           unitsCollection = world.get('unitsCollection')
           unitSelector ?= new Langtrainer.LangtrainerApp.Views.UnitSelector(
-            collection: unitsCollection
-            model: world.get('unit')
+            collection: Langtrainer.LangtrainerApp.currentUser.getCurrentCourse().get('unitsCollection')
+            model: Langtrainer.LangtrainerApp.currentUser.getCurrentCourse().getCurrentUnit()
           )
           unitSelector
             .render()

            .$el.detach().appendTo(languageSelectorContainer)

        if stepViewContainer?
          stepView ?= new Langtrainer.LangtrainerApp.Views.StepView(model: world.get('step'))
          stepView.$el.detach().appendTo(stepViewContainer)
          stepView.render()


        nativeLanguageSelectorContainer = $('.native-language-selector-container').first()
        if nativeLanguageSelectorContainer?
          nativeLanguagesCollection = world.get('nativeLanguagesCollection')
          nativeLanguageSelector ?= new Langtrainer.LangtrainerApp.Views.NativeLanguageSelector(
            collection: nativeLanguagesCollection
            model: world.get('nativeLanguage')
            label: 'native_language'
