Fixed another mistake I made