Fix unknown class reference