small clean up

Signed-off-by: Bruno Gaspar <brunofgaspar1@gmail.com>