Bugfix

Signed-off-by: constpetrov <constant.petrov@gmail.com>