bugfix in unit test