Fix test case for grammar addition