Повздорив с командованием, Буниятов был направлен в штрафную роту. В январе 1945 года его вызвал к себе комдив и предложил стать командиром штрафной роты. Капитан Буниятов был назначен командиром приданной 26 гв. ск 123-й отдельной штрафной роты 5-й ударной армии (1-й Белорусский фронт). 14 января в ходе Висло-Одерской операции в бою у селения Пальчек рота под командованием капитана Буниятова выполнила важную боевую задачу. Сам Буниятов позже вспоминал: