Fix build error

This reverts commit d753906.