Forgot a code change thing

Whoops.