Fix: print for the test function