small change
fixed problem with current button