Fixed display after error