[BUGFIX] Fix failing unit test