Simple Bug fixes