Bug fixes!