[Hotfix] Use proper variable