Hotfix a small issue