Осенью 1956-го года Надежда Кошеверова, поставившая знаменитую «Золушку» по сценарию Шварца, предложила драматургу экранизировать его пьесу «Тень». Но снова обращаться к известной пьесе, погружаться в хорошо освоенный материал — это было не так просто. Шварц решил переработать другую свою пьесу, не поставленную и не опубликованную, — «Голый король». Теперь сценарий стал называться иначе: «Два друга, или Раз, два, три!..». В последние годы Шварц много и часто болел, соответствовать ленфильмовским срокам не удавалось. Драматург умер в начале 1958-го года, не успев написать свой сценарий.
После его смерти вдова Шварца Екатерина Ивановна перепечатала рукопись сценария «Два друга». Получилось около сорока машинописных страниц, — вероятно, треть задуманного текста. Чтобы завершить сценарий, студия пригласила Николая Эрдмана. В архиве внучки Е. Л. Шварца Марии Олеговны Крыжановской хранится машинописный вариант сценария «Два друга», — вероятно, этот же текст оказался в руках Эрдмана, который, используя первые черновые сцены, должен был сделать законченный сценарий. Написанные Шварцем сцены Эрдман сильно сократил. Необходимо было слегка приглушить в них особенный «шварцевский» элемент — поскольку Эрдман всё равно не мог бы продолжить сценарий в «шварцевской» манере. Писали, что «Каин XVIII» — это «суп из топора» (первоначальный вариант полностью растворился в готовом сценарии) и что Шварц с Эрдманом «как масло и вода», их нельзя смешивать. Известный фильм Н. Н. Кошеверовой и М. Г. Шапиро говорит об обратном: смешивать можно.
Эрдман Николай, Шварц Евгений/ Николай Эрдман. «Киносценарии» // Блог Сеанса