Fix mistakes from last commit

Oops.