Got test to work

Couldn't make it all on one line.