Misc. fixes