fix minor type mismatch
minor update and fixes