Bug fix

Signed-off-by: selmeci <selmeci.roman@gmail.com>