Got two different types of tests working