Fix bug

Fix bug