Fixed a small bug in main test file.