fix(tests): Setting up tests