Bug fixes

Diego