module Sidekiq::SortedEntry::UniqueExtension

def delete
  SidekiqUniqueJobs::Unlockable.delete!(item) if super
  item
end

def remove_job
  super do |message|
    SidekiqUniqueJobs::Unlockable.delete!(Sidekiq.load_json(message))
    yield message
  end
end