Bug fixes

Some bugs…