Add proper dependency