Revert "unit test fix"

This reverts commit 2dbe5eb8aa306c1267cef7d19a440c357b07c9db.