fixing small import bug