Also updated the interface to prevent compiler warning.

Silly oversight on my part. Apologies for needing two commits/pulls for this.