fixing class in build script