Bug fixes..