Fixed a bug that prevented display from updating correctly.