Actually added unit test