Fixed a booboo I made