Fix that darn test