small bug fix
small change