Fix building failure