Add missing import.

This fixes Issue #34