Fixing update logic for HUD