code cleanup

Change-Id: I8203997f7d4052ed4ec3c937316aea5874239c27