Fix dependency for testing