Use correct methods in some tests

Seems like there was some "copy-paste" ;)