Helper for unit tests. Will remove once I get forward with unit tests.