Revert "fix build break"

This reverts commit 584ff2ac7471f63de05024446d94682a73a3dfbb.