Bug fixes.... tired...