Hotfix/broken unit test

Hotfix/broken unit test