Fix missing dependency issue