пара мелких изменений

потестил, должно работать