Added a hacky unit test