Fix production error