Added TODO comment for potential simplification.