Fix class ref

Change-Id: Ic12553496847b27a96bf32531261f046f3dd57d0
Merge "Fix class ref"