More misc. bug fixes