Add the correct dependency