cleaned up the tests to not store the stub in a secondary variable