Fix regression bug causing stage to lose focus