Update to fix compile error

Update to fix compile error