small fixes

small bugs