[Frontend-STFL] Improve wrapping performance.

Don't recalculate information we already know.