HOTFIX: Fix failing test