Fix bug: not mapping test type correctly