fixing small bug
stuff