Oops, forgot to fix this before last PR