diff --git a/apis/process/task.go b/apis/process/task.go
index 3dd3b03cdb55f8f66ca5a96435b9a81f8052e2b1..1ae502038ff1b58d7a809e779f49e89f1e3b07bd 100644
--- a/apis/process/task.go
+++ b/apis/process/task.go
@@ -48,7 +48,7 @@ func TaskList(c *gin.Context) {
 		file := make(map[string]interface{})
 		fileClassify := strings.Split(fn, ".")
 		fileDetails := strings.Split(fileClassify[0], "-")
-		switch fileClassify[1] {
+		switch fileClassify[len(fileClassify)-1]] {
 		case "py":
 			file["classify"] = "Python"
 		case "sh":