Fix reference to right class and namespace name