Bug fixes ;)