Fixes to refactor

Отрефакторил.