Small bug fix

Fixed a small dependency bug