Fix a bug in #5