Updated to fix compilation error.