Quick repair Graphics unit

Sorry Krishny - we managed to break the end of the Graphics unit when we updated it.