Fixes the dependancy for unit tests