TST/FIX: correct assertion error
CLN: remove print stmts