Fix webshot code