Code cleanup. Small change.