Fixed things i forgot ;)