Missing import, fix name

(this was being missed by tests).
Merged