Fix tests

Oops.
1 file changed