Add tags test and fix wrong import