Fixed reference to $di