Small misc. fixes