Bring it in-line with demo coding style.
Forgot to add extra argument. Fix that.