Simplify test-case.
Fix return of print.