fix code to build.