Updating last bit of test coverage
More testing setup