Fix building issue