Got test to work