Fixed compile error with new version