Minor class fix.