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

Fix two broken tests due to remove of debug statements