Fix a bug in test code #31