Bugfix unit test