fixes last unit test