Fix dependency issue