Fixed dumb mistake.

(still not working fully)