Fix building problem