Fix small dependency problem