Moved test model into closer to the testcase