Small fixes

Signed-off-by: hajee <hajee@moretIA.com>