some more polishing

Signed-off-by: loopasam <samuel.croset@gmail.com>