Fixing bug

Signed-off-by: Ventsislav Dimitrov <wencislawdimitrow@gmail.com>