Закрытие настраиваемого диалогового окна предупреждения при нажатии кнопки

У меня проблемы с закрытием диалогового окна с предупреждениями. Я использую инфлятор макета для создания диалогового окна, поэтому я не уверен, как бы я закрыл его после того, как закончу с ним. Код ниже:

AlertDialog.Builder dialog = new AlertDialog.Builder(AddData.this);
DialogInterface dia = new DialogInterface();

//Create a custom layout for the dialog box
LayoutInflater inflater = (LayoutInflater)AddData.this.getSystemService(LAYOUT_INFLATER_SERVICE);
View layout = inflater.inflate(R.layout.add_rep_1_set, parent, false);

TextView title = (TextView)layout.findViewById(R.id.rep_1_title);
Button add_item = (Button)layout.findViewById(R.id.add_button);

add_item.setOnClickListener(new OnClickListener()
{
        @Override
        public void onClick(View v)
        {
        //Need this to close the alert dialog box
        }
});

title.setText(workout_items[position]);
dialog.setView(layout);
dialog.show();

Я не могу вызвать finish, потому что это закрывает список, из которого запускается диалоговое окно с предупреждением, и вызовы dialog.dismiss недоступны для меня.

Как вы думаете, что мне нужно сделать, чтобы это исправить?