Remove one more TODO