Fixing up deprecation warning.