fixes small bugs