Fixed a bug with wrong display functions