Fix local import