Add back removed dependency