Update for correct display