Small display fix