Fix imports and module name.