Removed broken test code
Fixed bug, prob introduced by sloppy cut and paste
Docstring update