Fix a bug line 26