General small fixes

Signed-off-by: Vlad Velici <vlad.velici@gmail.com>