Didnt need to setup testing in that way - the original way worked.