Fix incorrect dependencies

Fixes #6