small bug fix for display