add needed import
Merge pull request #3190 from dimagi/cz

add needed import