Fix the setup.py script not import cms

Fixes #1651.