diff --git a/modules/auth/cloudbrain.go b/modules/auth/cloudbrain.go
index 277588f40..cfa1e81c3 100755
--- a/modules/auth/cloudbrain.go
+++ b/modules/auth/cloudbrain.go
@@ -13,6 +13,8 @@ type CreateCloudBrainForm struct {
 	JobType              string `form:"job_type" binding:"Required"`
 	BenchmarkCategory    string `form:"get_benchmark_category"`
 	GpuType              string `form:"gpu_type"`
+	TrainUrl             string `form:"train_url"`
+	TestUrl              string `form:"test_url"`
 	ResourceSpecId       int    `form:"resource_spec_id" binding:"Required"`
 	BenchmarkTypeID      int    `form:"benchmark_types_id"`
 	BenchmarkChildTypeID int    `form:"benchmark_child_types_id"`