Bug fixes (2)