Missed test refactoring

Signed-off-by: Luca Degasperi <luca@lucadegasperi.com>