Very simple patch.
Just update.

Tested to compile and run.