Cleanup assign_happens_before.