Fix Exception in test