This test feels like overkill, but anyways, thoughts on a better way to test this?