De 158e werd geselecteerd als een deel van de speerpunt van de invasie van Japan en gepland om Japanse luchtwaarschuwingsstations ten zuiden van Kyushu te laten zwijgen twee dagen voor de invasie. De geplande invasie van Japan werd nooit uitgevoerd; na de overgave van Japan
, werd de eenheid op 13 oktober 1945 naar Yokohama, Japan, gestuurd als onderdeel van het bezettingsleger van de Verenigde Staten.