Reintroduced required import and silenced PyFlakes warning