Fix build problem