Minor code simplification in tests

Signed-off-by: Francis Galiegue <fgaliegue@gmail.com>
One minor code modification

Signed-off-by: Francis Galiegue <fgaliegue@gmail.com>