Bugfix in test