Fix issue with missing dependency