Fix for display function