Change to use proper dependency