как установить прослушиватель onclick для кнопки изображения в alerttdialog

У меня есть макет с ImageButton, который раздувается в AlertDialog, где / как мне установить прослушиватель onClick?

Вот код, который я пытался использовать:

    ImageButton ib = (ImageButton) findViewById(R.id.searchbutton);
    ib.setOnClickListener(new View.OnClickListener() {
        @Override
        public void onClick(View v) {
            Toast.makeText(TravelBite.this, "test", Toast.LENGTH_SHORT).show();
        }
    });