Fixed up broken test
First test is looking good