Tiny refactoring.

Signed-off-by: Mariusz Fik <fisiu@fidano.pl>