Fix for build error