Cuando el Shah declaró la ley marcial en respuesta a las crecientes protestas en 1978, y puso los oficiales militares a cargo, Yahanbaní no era uno de los comandantes militares desde que tenía muy poca experiencia con los asuntos de seguridad interna. Como resultado, cuando el Sha huyó, a pesar de la insistencia de su familia, a sus amigos en la Fuerza Aérea de los EE. UU., así como el mismo y su hija Shahnaz (que era su hermana en la ley), Yahanbaní falsamente creía que él estaba a salvo de posibles purgas y represalias contra los agentes de seguridad que reprimieron las protestas, así como su creencia de que la fuerza aérea de gran alcance que Irán tuviera sería un testimonio de su lealtad al país, no al propio Shah. Él se equivocó cuando el ayatolá Jomeini volvió y ordenó a la Guardia Revolucionaria de arrestarlo en la sede de la fuerza aérea. Fue uno de los primeros de los generales del Sha en ser detenido y fue enviado a una carrera judicial por Sadeq Jaljalí.