Bug fixes

Bug fixes!