Fix ui module build.