Fixed minor oversight in test program.