fix button assignment