fixes som bugs