diff --git a/src/main/kotlin/com/test/service/TestService.kt b/src/main/kotlin/com/test/service/TestService.kt
index 78f3959..fa66039 100644
--- a/src/main/kotlin/com/test/service/TestService.kt
+++ b/src/main/kotlin/com/test/service/TestService.kt
@@ -20,7 +20,7 @@ class TestService(private val testEntityRepository: TestEntityRepository) {
 
     fun update(model: TestModel): TestModel {
         var entity = testEntityRepository.save(TestEntity(model.id, model.key, model.value))
-        return TestModel(null, entity.key, entity.value)
+        return TestModel(model.id, entity.key, entity.value)
     }
 
     fun get(id: Long): TestModel? {