Add missing close import