The government is being urged to extend bereavement payments to co-habiting couples after a court ruling.