fix(test): Updated should and made changes to testing setup