Inconsistent hilighting when using Treesitter for keywords

[baco]

• [x] I have carefully read and followed the Installation section of the README
• [x] I have searched through the closed issues to see if there is an existing solution for my issue

Terminal Emulator Info

Terminal emulator(s) and their version(s) used, if applicable: Kitty Terminal, Tilix and Neovim-GTK

Output From vim --version

NVIM v0.6.0-dev+105-g8baf7bce2
Build type: RelWithDebInfo
LuaJIT 2.1.0-beta3
Compilation: /usr/bin/gcc-11 -U_FORTIFY_SOURCE -D_FORTIFY_SOURCE=1 -DNVIM_TS_HAS_SET_MATCH_LIMIT -O2 -g -Og -g -Wall -Wextra -pedantic -Wno-unused-parameter -Wstrict-prototypes -std=gnu99 -Wshadow -Wconversion -Wmissing-prototypes -Wimplicit-fallthrough -Wvla -fstack-protector-strong -fno-common -fdiagnostics-color=always -DINCLUDE_GENERATED_DECLARATIONS -D_GNU_SOURCE -DNVIM_MSGPACK_HAS_FLOAT32 -DNVIM_UNIBI_HAS_VAR_FROM -DMIN_LOG_LEVEL=3 -I/home/runner/work/neovim/neovim/build/config -I/home/runner/work/neovim/neovim/src -I/home/runner/work/neovim/neovim/.deps/usr/include -I/usr/include -I/home/runner/work/neovim/neovim/build/src/nvim/auto -I/home/runner/work/neovim/neovim/build/include
Compiled by runner@fv-az87-30

Features: +acl +iconv +tui
See ":help feature-compile"

   system vimrc file: "$VIM/sysinit.vim"
  fall-back for $VIM: "/share/nvim"

Run :checkhealth for more info

Issue Description

When turning on Treesitter's syntax highlighting Python's statements (pythonStatement) are interpreted as Keywords, and since pythonStatement is set to purple and Keywords are set to red the onedark theme gets a rather drastic change:

This is the theme normally without Treesiter's highlighting:

After turning on Treesitter's highlighting

it drastically affects every keyword: def, class, return, if, for, while, among others.