diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index ef21d2a..0aa0a32 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -7,11 +7,9 @@ add_library(compiler
         dialect/onnx/onnx_ops.hpp
         dialect/krnl/krnl_helper.cpp
         dialect/krnl/krnl_helper.hpp
-        pass/shape_inference_pass.cpp
         pass/shape_inference_interface.hpp
         dialect/onnx/onnxop.inc
         pass/onnx_combine.cpp
-        pass/lower_frontend_to_krnl.cpp
         pass/passes.hpp)
 
 # Include root src directory.
diff --git a/src/tool/onnf_opt/CMakeLists.txt b/src/tool/onnf_opt/CMakeLists.txt
index 7cde1f2..0d0b302 100644
--- a/src/tool/onnf_opt/CMakeLists.txt
+++ b/src/tool/onnf_opt/CMakeLists.txt
@@ -4,9 +4,6 @@ add_dependencies(onnf-opt gen_krnl_ops)
 target_include_directories(onnf-opt PRIVATE ${ONNF_SRC_ROOT})
 target_include_directories(onnf-opt PRIVATE ${ONNF_BIN_ROOT})
 
-target_link_libraries(onnf-opt compiler ${MLIRLibs})
+target_link_libraries(onnf-opt ${MLIRLibs} curses)
 whole_archive_link_mlir(onnf-opt ${MLIRWholeArchiveLibs})
-whole_archive_link_onnf(onnf-opt onnf_transform onnf_lower_frontend onnf_shape_inference)
-
-# TODO: need to investigate how to whole-archive link compiler pass to onnf-opt.
-target_link_libraries(onnf-opt compiler)
+whole_archive_link_onnf(onnf-opt compiler onnf_transform onnf_lower_frontend onnf_shape_inference)
\ No newline at end of file