Fix a bug

Fix a bug