Fixed the errors for the test program