Revert "unit test fix"

This reverts commit 28fc8714ee2d75e2ffaee91c90e456cbf2a2ae10.