Adding in test protection