Might need to write test helpers for asserts