Class fix
Merge branch 'hotfix/hotfix'