Bugfix in unittest