fixed #37

Исправляем код согласно рекомендации Ramir.