Fix two broken tests due to remove of debug statements
Merge pull request #459 from jamie-cyber1/fix_unit

Fix two broken tests due to remove of debug statements