Fix an import for #61