Bug fixes, bug fixes!