fix import issue with categorical variables
Reviewer: Dan Jao
Issue: none