diff --git a/kernel/api/riff.go b/kernel/api/riff.go
index 575e34860..7a3cb31a2 100644
--- a/kernel/api/riff.go
+++ b/kernel/api/riff.go
@@ -36,7 +36,7 @@ func getTreeRiffCards(c *gin.Context) {
 		return
 	}
 
-	rootID := arg["rootID"].(string)
+	rootID := arg["id"].(string)
 	page := int(arg["page"].(float64))
 	blockIDs, total, pageCount := model.GetTreeFlashcards(rootID, page)
 	ret.Data = map[string]interface{}{
@@ -55,7 +55,7 @@ func getRiffCards(c *gin.Context) {
 		return
 	}
 
-	deckID := arg["deckID"].(string)
+	deckID := arg["id"].(string)
 	page := int(arg["page"].(float64))
 	blockIDs, total, pageCount := model.GetFlashcards(deckID, page)
 	ret.Data = map[string]interface{}{