Revert "Fixed unittest speed"

This reverts commit 4f8d6bfe4ab7732f8312a7471110ab592b7c89c6.