I’ve run into similar issues before where things behave differently in headless mode. One trick that sometimes helps is setting a fixed window size options.add_argument("--window-size=1920,1080")
before launching the driver. Some elements don’t load properly in headless mode unless the viewport is large enough.