Actually fix the previous bug. Whoops.