diff --git a/app/src/main/java/dev/lonami/talaria/TalariaApp.kt b/app/src/main/java/dev/lonami/talaria/TalariaApp.kt
index a6efd88..648044d 100644
--- a/app/src/main/java/dev/lonami/talaria/TalariaApp.kt
+++ b/app/src/main/java/dev/lonami/talaria/TalariaApp.kt
@@ -13,9 +13,9 @@ import androidx.navigation.compose.composable
 import androidx.navigation.compose.currentBackStackEntryAsState
 import androidx.navigation.compose.rememberNavController
 import dev.lonami.talaria.bindings.Native
-import dev.lonami.talaria.ui.ChatScreen
-import dev.lonami.talaria.ui.DialogScreen
-import dev.lonami.talaria.ui.LoginScreen
+import dev.lonami.talaria.ui.screens.ChatScreen
+import dev.lonami.talaria.ui.screens.DialogScreen
+import dev.lonami.talaria.ui.screens.LoginScreen
 
 enum class TalariaScreen(@StringRes val title: Int) {
     Login(title = R.string.app_name),
diff --git a/app/src/main/java/dev/lonami/talaria/ui/state/DialogViewModel.kt b/app/src/main/java/dev/lonami/talaria/ui/state/DialogViewModel.kt
index db911c2..6393c35 100644
--- a/app/src/main/java/dev/lonami/talaria/ui/state/DialogViewModel.kt
+++ b/app/src/main/java/dev/lonami/talaria/ui/state/DialogViewModel.kt
@@ -1,13 +1,13 @@
 package dev.lonami.talaria.ui.state
 
 import androidx.lifecycle.ViewModel
-import dev.lonami.talaria.data.DialogSource
+import dev.lonami.talaria.data.DialogRepository
 import dev.lonami.talaria.data.NativeDialogRepository
 import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.StateFlow
 import kotlinx.coroutines.flow.asStateFlow
 
-class DialogViewModel(private val repository: DialogSource = NativeDialogRepository()) :
+class DialogViewModel(private val repository: DialogRepository = NativeDialogRepository()) :
     ViewModel() {
     private val _uiState = MutableStateFlow(DialogUiState())
     val uiState: StateFlow<DialogUiState> = _uiState.asStateFlow()