bug fix

Signed-off-by: Wadjetz <egor.neon@gmail.com>