Fix not compile
Revert "Fix not compile"

This reverts commit 415d9eb061e787620874dca334d68cd40afd88a8.