Correct generator test case

Signed-off-by: Gary Pamparà <gpampara@gmail.com>