Fix tests that were using wrong method.