More fixes

Signed-off-by: kecinzer <kecinzer@gmail.com>