Remove debug prints
Fix thrown exception in unit test