Fix tiny bug with the element generators