Small change to $digest