Fix import for 3.2