Implemented unit test helper function... kinda