Небольшой рефакторинг, fix #17