Update tests for new lang selection