Fix bug

Bug fixed