diff --git a/queries/python/highlights.scm b/queries/python/highlights.scm
index 9d9f8a5bb..9bcfa183d 100644
--- a/queries/python/highlights.scm
+++ b/queries/python/highlights.scm
@@ -195,17 +195,20 @@
   "del"
 ] @keyword.operator
 
+[
+  "def"
+  "lambda"
+] @keyword.function
+
 [
   "assert"
   "async"
   "await"
   "class"
-  "def"
   "except"
   "exec"
   "finally"
   "global"
-  "lambda"
   "nonlocal"
   "pass"
   "print"