diff --git a/src/main/kotlin/ru/itmo/dws/itmotrip/model/LocationType.kt b/src/main/kotlin/ru/itmo/dws/itmotrip/model/LocationType.kt
index 0856589..71b3e21 100644
--- a/src/main/kotlin/ru/itmo/dws/itmotrip/model/LocationType.kt
+++ b/src/main/kotlin/ru/itmo/dws/itmotrip/model/LocationType.kt
@@ -1,6 +1,11 @@
 package ru.itmo.dws.itmotrip.model
 
+import org.springframework.data.annotation.Id
+import org.springframework.data.relational.core.mapping.Table
+
+@Table("location_types")
 data class LocationType(
+    @Id
     val id: Long,
     val code: String
 )
\ No newline at end of file
diff --git a/src/main/kotlin/ru/itmo/dws/itmotrip/model/TransportType.kt b/src/main/kotlin/ru/itmo/dws/itmotrip/model/TransportType.kt
index c1c3195..c89f7fc 100644
--- a/src/main/kotlin/ru/itmo/dws/itmotrip/model/TransportType.kt
+++ b/src/main/kotlin/ru/itmo/dws/itmotrip/model/TransportType.kt
@@ -1,6 +1,11 @@
 package ru.itmo.dws.itmotrip.model
 
+import org.springframework.data.annotation.Id
+import org.springframework.data.relational.core.mapping.Table
+
+@Table("location_types")
 data class TransportType (
+    @Id
     val id: Long,
     val logoUrl: String?,
     val nameRu: String