Fixed error in import with @jurgenvinju