Slight code clearup.

Signed-off-by: ashwinh <ashwinh@cybage.com>