Policy test working again

Closes #6