Hotfix for keyerror