Fixed a small import bug