bug fixes

gerhjklsegrhkluegrshjklegrshljk