Fixed minor import bug