Reference the right classes