Revert "To fix the build"

This reverts commit 5663b2cf443b2be850c964da8598088455d6bb9b.