diff --git a/app/src/main/java/com/pomo/myapplication/timerroom/TaskDao.kt b/app/src/main/java/com/pomo/myapplication/timerroom/TaskDao.kt
index 8810a7c55bab86b05f0513b3468ad4d94b38f3c5..e96f220f75ef05fbec84587a28cffb50f4bf30df 100644
--- a/app/src/main/java/com/pomo/myapplication/timerroom/TaskDao.kt
+++ b/app/src/main/java/com/pomo/myapplication/timerroom/TaskDao.kt
@@ -5,18 +5,19 @@ import androidx.room.Delete
 import androidx.room.Insert
 import androidx.room.Update
 import androidx.room.*
+import kotlinx.coroutines.flow.Flow
 
 @Dao
 interface TaskDao {
     @Query("SELECT * FROM tasks")
-    suspend fun getAllTasks(): List<Task>
+    fun getAllTasks(): Flow<List<TaskEntity>>
 
-    @Insert(onConflict = OnConflictStrategy.REPLACE)
-    suspend fun insertTask(task: Task)
+    @Insert
+    suspend fun insertTask(task: TaskEntity)
 
     @Delete
-    suspend fun deleteTask(task: Task)
+    suspend fun deleteTask(task: TaskEntity)
 
     @Update
-    suspend fun updateTask(task: Task)
+    suspend fun updateTask(task: TaskEntity)
 }