Update to use prod