Fix error class reference