Fix spec that failed after inner class fix