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 8c41817..e63dba2 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
@@ -1,15 +1,17 @@
 package com.example.kocelainterview.data.local_data_source
 
 import androidx.room.Entity
+import androidx.room.PrimaryKey
 import com.example.kocelainterview.data.remote.dto.ShipDto
 
 @Entity(tableName = "ships")
 data class ShipEntity(
-   // @PrimaryKey val id:Int,
+
     val ship_id: String,
     val active: Boolean,
     val image: String,
     val ship_name: String,
+    @PrimaryKey
     val weight_kg: Int,
     val year_built: Int
 )