One act, regardless of how heroic it might be, doesn't mean that the person should be free from consequences for the rest of their lives.

The US extraditing him for conspiracy charges would be really tragic. Sweden wanting to hold him accountable for his behavior afterwards is completely fine.