Fix test
Little tweak in parser