Mała refaktoryzacja -> zwiększone pokrycie kodu testami

Signed-off-by: mateusz <kontakt@mateuszmidor.com>