diff --git a/app/src/main/java/com/example/notatkon/dao/NoteDao.java b/app/src/main/java/com/example/notatkon/dao/NoteDao.java
index d0dfed6..2f1f1ee 100644
--- a/app/src/main/java/com/example/notatkon/dao/NoteDao.java
+++ b/app/src/main/java/com/example/notatkon/dao/NoteDao.java
@@ -1,4 +1,4 @@
-package com.example.notatkon.database;
+package com.example.notatkon.dao;
 
 import androidx.lifecycle.LiveData;
 import androidx.room.Dao;
@@ -7,6 +7,8 @@ import androidx.room.Insert;
 import androidx.room.OnConflictStrategy;
 import androidx.room.Query;
 
+import com.example.notatkon.entities.NoteEntity;
+
 import java.util.List;
 
 @Dao
diff --git a/app/src/main/java/com/example/notatkon/entities/NoteEntity.java b/app/src/main/java/com/example/notatkon/entities/NoteEntity.java
index 366c529..6d51903 100644
--- a/app/src/main/java/com/example/notatkon/entities/NoteEntity.java
+++ b/app/src/main/java/com/example/notatkon/entities/NoteEntity.java
@@ -1,4 +1,4 @@
-package com.example.notatkon.database;
+package com.example.notatkon.entities;
 
 import androidx.annotation.NonNull;
 import androidx.room.ColumnInfo;