Fixed another mistake