Fixed display timeout problem