Fixes a small class mistake.