Update: Fix screen resize