[#1117] Fix an import (PEP8)