Forgot to make equivalent changes to test that were done on deprecation notice