Fixes a class displaying error