Fixes a deprecation warning.

Co-authored-by: Jarek Radosz <jradosz@gmail.com>