Fix a bug

Fix a bug.