fix bug for production