Update test to not throw error