Moved unit tests to integration tests