Fix for build problem