Minor refactor

Signed-off-by: Ladislav Klenovic <klenovic@nerdrobot.net>