FIX: Fixed setting of display