-- Bug fixes