Fix missing status KeyError issue