remove test code in main function

Minor change.