fix second bug