The Supreme Court's ruling in the case said exactly this and the opinions of some justices implied the government should pursue such charges post publication.
I imagine that Assange taking active steps to help gather the classified information instead of just publishing material gathered independently and leaked is what changed the math in this case.