small fixes

I made some small fixes