Code cleanup for your pull request

Это моя правка, осталась от еще незапушеного комита.