Fix import
Merge pull request #4 from pmorie/cmd

Fix import