UKSM: Cleanups

Signed-off-by: Timofey Titovets <nefelim4ag@gmail.com>