Typo
Added some code that was missing.
Work continues on changing stuff to use the new graphics classes.