Fix deprecation warning on prod