Fix success check