forgot testDataset

Signed-off-by: Mauro Scomparin <scompo@gmail.com>