A quick update to fix the overlay