Bugfix for unit tests.