Fix a bug in test code #4