Fix dep issue

Proper fix the dep issue