Add missing future import