Distribution Awareness for AI System Testing