[BUGFIX] Fix failing unittest