Fix missing import

See #71.