Add missing optional dependency