Fix import and method call preventing the build script from succeeding