Remove incorrect class ref