Minor code cleanup

Signed-off-by: Claudio Matsuoka <cmatsuoka@gmail.com>