Bug fixes bro