Update test case to standardise errors