Fix ImportError introduced by removing the variable in 7d914a0533f6c07879628802347f6800236350c8