Fixing the small bug I noted.