Remove un-used import, thanks pylint