Update to fix compile error

Update to fix compile error
Merge pull request #322 from EliteEng/dev

Update to fix compile error