Re-instate missing test I accidentally removed