[CLEANUP] Compilation Fix for previous commit