Added import forgotten in 2ee21d9.

Thanks ratsonale for spotting this mistake.