Fix labels and add better tests.

Signed-off-by: Yoan Blanc <yoan.blanc@cern.ch>