Fix

Signed-off-by: Mario Kozjak <kozjakm1@gmail.com>