FIX: Minor test fix