OOops. Fixed incomplete refactoring from earlier.
Small fix readme