Small fix to compile with latest 2.0beta SDK