HOTFIX: Fix broken unit test