Finish unit test on policy-models