Updated Policy tests to test Driver instead