This import should be in a section on its own above those for selenium (because it is a standard library module).