Added a very basic omni test as we didn't cover it previously