More cleanup

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