Misc. bug-fixes