Missing import.

Fixes #269