unit test and stubs

stubs need altering for the test to work