Fix F401 error (module imported but unused)

[ci skip]