Rejigger that code.