Fixed unit tests to match code change.
Merge pull request #238 from scrollback/testscriptfix

Fixed unit tests to match code change.