unit test bugfix