diff --git a/app/src/main/java/com/example/kocelainterview/data/local_data_source/ShipEntity.kt b/app/src/main/java/com/example/kocelainterview/data/local_data_source/ShipEntity.kt
index 45030b5..24db52f 100644
--- a/app/src/main/java/com/example/kocelainterview/data/local_data_source/ShipEntity.kt
+++ b/app/src/main/java/com/example/kocelainterview/data/local_data_source/ShipEntity.kt
@@ -6,9 +6,10 @@ import com.example.kocelainterview.data.remote.dto.ShipDto
 
 @Entity(tableName = "ships")
 data class ShipEntity(
-    @PrimaryKey val ship_id: String,
+    @PrimaryKey val id:Int,
+    val ship_id: String,
     val active: Boolean,
-    val image: String?,
+    val image: String,
     val ship_name: String,
     val weight_kg: Int,
     val year_built: Int