Update to match changes to autoflake