Fixed git derp of missing piece code