Bugfix for test