[BUGFIX] Avoid exception for int label column