Fixing the test program