* Small refactoring

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