Small fixes

fix it fix it fix it fix it...