Bug fixes xD