Fix version causing import error