Added missing dependency
Test fixed