Add missing tests

This is the second time I've seen the command broken even though the tests passed. The missing tests should be added

changed milestone to %0.4.0