got the tests working