Fixing the build error