"Эпизод" is ok.

For 'act' as in a play, we use both "акт" и "действие". E.g. "пьеса в трёх действиях" (a three-act play). "Пятиактная трагедия" (A five-act tragedy).