Again bad boy, removed unused import (flake8)