refactor(tests): Unit testing refactored to work