Check which exception is returned in tests, modify tests