Bug fixes. Oops!