diff --git a/server/model/agent.go b/server/model/agent.go
index 26b72f4ab..c268644c1 100644
--- a/server/model/agent.go
+++ b/server/model/agent.go
@@ -25,7 +25,7 @@ type Agent struct {
 	Platform    string `json:"platform"      xorm:"VARCHAR(100) 'platform'"`
 	Backend     string `json:"backend"       xorm:"VARCHAR(100) 'backend'"`
 	Capacity    int32  `json:"capacity"      xorm:"capacity"`
-	Version     string `json:"version"       xorm:"version"`
+	Version     string `json:"version"       xorm:"'version'"`
 	NoSchedule  bool   `json:"no_schedule"   xorm:"no_schedule"`
 } //	@name Agent
 
diff --git a/server/store/datastore/agent_test.go b/server/store/datastore/agent_test.go
index 992412128..f08a5ee74 100644
--- a/server/store/datastore/agent_test.go
+++ b/server/store/datastore/agent_test.go
@@ -87,3 +87,22 @@ func TestAgentList(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, 1, len(agents))
 }
+
+func TestAgentUpdate(t *testing.T) {
+	store, closer := newTestStore(t, new(model.Agent))
+	defer closer()
+
+	agent := &model.Agent{
+		ID:    int64(1),
+		Name:  "test",
+		Token: "secret-token",
+	}
+	err := store.AgentCreate(agent)
+	assert.NoError(t, err)
+
+	agent.Backend = "local"
+	agent.Capacity = 2
+	agent.Version = "next-abcdef"
+	err = store.AgentUpdate(agent)
+	assert.NoError(t, err)
+}