Fixed unit testing

Signed-off-by: Julien Bonvarlet <jbonva@gmail.com>