Final fixes I hope

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