Morale check after step losses from assualt

[stodge]

After an assault, do I perform an M2 morale check on all units remaining after step losses are applied? Or is the M2 morale check only for the units that took step losses and survived? I think it’s the latter but I just wanted to check.

Thanks