Add argument to test element_completion