* Small bug fix

Signed-off-by: Jiri Hnidek <jiri.hnidek@tul.cz>